package com.study.zk.lock.zk;

import org.I0Itec.zkclient.IZkDataListener;
import org.I0Itec.zkclient.ZkClient;
import org.I0Itec.zkclient.exception.ZkNodeExistsException;

import java.util.Collections;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;

/**
 * 解决第一个版本的惊群效应问题
 * <p>
 * 利用临时顺序节点来实现分布式锁
 * 获取锁：排队取号（创建自己的临时顺序节点），然后判断自己是否是最小号；如是，则获得锁。
 * 释放锁：删除自己创建的临时顺序节点
 *
 * todo 当前 lockPath 还不能用于分布式（或多线程）中，因为类型是private；还不能确定线程抢的是不是同一把分布式锁。
 *
 * todo 实现锁的可重入：新增一个计数功能？
 *
 * todo 同一把锁，每次获取加一，释放则减一。减到0时即可把节点删除
 *
 * todo ZookeeperDistributedLock中使用ThreadLocal变量定义方式，保证了多线程下抢的是同一把锁beforePath？
 *
 * @author: chenhs
 * @date: Created in 15:13 2020/2/4
 **/
public class ZkDistributeImproveLock implements Lock {

    private String lockPath;

    private ZkClient client;

    private String currentPath;

    private String beforePath;

    public ZkDistributeImproveLock(String path) {
        this.lockPath = path;
        this.client = new ZkClient("localhost:2181");
        this.client.setZkSerializer(new MyZkSerializer());
        //创建一个根节点
        if (!this.client.exists(lockPath)) {
            try {
                this.client.createPersistent(lockPath);
            } catch (ZkNodeExistsException e) {

            }
        }
    }

    /**
     * 不会阻塞
     *
     * 如果一个线程取到号排队过程退出，那么它后面一个节点序号会变成当前最小节点序号；
     *
     * @return
     */
    @Override
    public boolean tryLock() {
        if (this.currentPath == null) {
            //创建临时顺序节点，存储任意值
            currentPath = this.client.createEphemeralSequential(lockPath + "/", "xxx");
        }
        //获得所有子节点
        List<String> children = this.client.getChildren(lockPath);
        //排序list
        Collections.sort(children);
        //判断当前是否最小
        if (currentPath.equals(lockPath + "/" + children.get(0))) {
            return true;
        } else {
            //得到子节点的索引号
            int curIndex = children.indexOf(currentPath.substring(lockPath.length() + 1));
            //取到前一个子节点
            beforePath = lockPath + "/" + children.get(curIndex - 1);
        }
        return false;
    }

    /**
     * 如果获取不到锁，阻塞等待。
     * 哦，tryLock为true则获取锁，跳出递归。
     */
    @Override
    public void lock() {
        if (!tryLock()) {
            //没获得锁，阻塞自己
            waitForLock();
            //再次尝试
            lock();
        }
    }

    private void waitForLock() {
        CountDownLatch cdl = new CountDownLatch(1);

        IZkDataListener listener = new IZkDataListener() {
            @Override
            public void handleDataDeleted(String dataPath) throws Exception {
                System.out.println("---------------监听到节点被删除");
                cdl.countDown();
            }

            @Override
            public void handleDataChange(String dataPath, Object data) throws Exception {
            }
        };
        //注册监听自己的前一个节点
        client.subscribeDataChanges(this.beforePath, listener);

        //阻塞自己
        if (this.client.exists(this.beforePath)) {
            try {
                cdl.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        //取消注册
        client.unsubscribeDataChanges(this.beforePath, listener);
    }

    @Override
    public void unlock() {
        this.client.delete(this.currentPath);
    }


    @Override
    public void lockInterruptibly() throws InterruptedException {

    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return false;
    }

    @Override
    public Condition newCondition() {
        return null;
    }

}
